package com.ruoyi.paipan.common;

import com.ruoyi.paipan.utils.LunarUtil;
import com.ruoyi.paipan.utils.ShouXingUtil;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

public class LunarYear {
    public static List<String> YUAN = Arrays.asList("下", "上", "中");
    public static List<String> YUN = Arrays.asList("七", "八", "九", "一", "二", "三", "四", "五", "六");
    private static List<Integer> LEAP_11 = Arrays.asList(75, 94, 170, 265, 322, 398, 469, 553, 583, 610, 678, 735, 754, 773, 849, 887, 936, 1050, 1069, 1126, 1145, 1164, 1183, 1259, 1278, 1308, 1373, 1403, 1441, 1460, 1498, 1555, 1593, 1612, 1631, 1642, 2033, 2128, 2147, 2242, 2614, 2728, 2910, 3062, 3244, 3339, 3616, 3711, 3730, 3825, 4007, 4159, 4197, 4322, 4341, 4379, 4417, 4531, 4599, 4694, 4713, 4789, 4808, 4971, 5085, 5104, 5161, 5180, 5199, 5294, 5305, 5476, 5677, 5696, 5772, 5791, 5848, 5886, 6049, 6068, 6144, 6163, 6258, 6402, 6440, 6497, 6516, 6630, 6641, 6660, 6679, 6736, 6774, 6850, 6869, 6899, 6918, 6994, 7013, 7032, 7051, 7070, 7089, 7108, 7127, 7146, 7222, 7271, 7290, 7309, 7366, 7385, 7404, 7442, 7461, 7480, 7491, 7499, 7594, 7624, 7643, 7662, 7681, 7719, 7738, 7814, 7863, 7882, 7901, 7939, 7958, 7977, 7996, 8034, 8053, 8072, 8091, 8121, 8159, 8186, 8216, 8235, 8254, 8273, 8311, 8330, 8341, 8349, 8368, 8444, 8463, 8474, 8493, 8531, 8569, 8588, 8626, 8664, 8683, 8694, 8702, 8713, 8721, 8751, 8789, 8808, 8816, 8827, 8846, 8884, 8903, 8922, 8941, 8971, 9036, 9066, 9085, 9104, 9123, 9142, 9161, 9180, 9199, 9218, 9256, 9294, 9313, 9324, 9343, 9362, 9381, 9419, 9438, 9476, 9514, 9533, 9544, 9552, 9563, 9571, 9582, 9601, 9639, 9658, 9666, 9677, 9696, 9734, 9753, 9772, 9791, 9802, 9821, 9886, 9897, 9916, 9935, 9954, 9973, 9992);
    private static List<Integer> LEAP_12 = Arrays.asList(37, 56, 113, 132, 151, 189, 208, 227, 246, 284, 303, 341, 360, 379, 417, 436, 458, 477, 496, 515, 534, 572, 591, 629, 648, 667, 697, 716, 792, 811, 830, 868, 906, 925, 944, 963, 982, 1001, 1020, 1039, 1058, 1088, 1153, 1202, 1221, 1240, 1297, 1335, 1392, 1411, 1422, 1430, 1517, 1525, 1536, 1574, 3358, 3472, 3806, 3988, 4751, 4941, 5066, 5123, 5275, 5343, 5438, 5457, 5495, 5533, 5552, 5715, 5810, 5829, 5905, 5924, 6421, 6535, 6793, 6812, 6888, 6907, 7002, 7184, 7260, 7279, 7374, 7556, 7746, 7757, 7776, 7833, 7852, 7871, 7966, 8015, 8110, 8129, 8148, 8224, 8243, 8338, 8406, 8425, 8482, 8501, 8520, 8558, 8596, 8607, 8615, 8645, 8740, 8778, 8835, 8865, 8930, 8960, 8979, 8998, 9017, 9055, 9074, 9093, 9112, 9150, 9188, 9237, 9275, 9332, 9351, 9370, 9408, 9427, 9446, 9457, 9465, 9495, 9560, 9590, 9628, 9647, 9685, 9715, 9742, 9780, 9810, 9818, 9829, 9848, 9867, 9905, 9924, 9943, 9962, 10000);
    private static LunarYear CACHE_YEAR = null;
    public static List<Integer> YMC = Arrays.asList(11, 12, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10);
    private int year;
    private int ganIndex;
    private int zhiIndex;
    private List<LunarMonth> months = new ArrayList<>();
    private List<Double> jieQiJulianDays = new ArrayList<>();

    public LunarYear(int year) {
        this.year = year;
        int offset = year-4;
        int yearGanIndex = offset % 10;
        int yearZhiIndex = offset % 12;
        if (yearGanIndex < 0) {
            yearGanIndex += 10;
        }
        if (yearZhiIndex < 0) {
            yearZhiIndex += 12;
        }
        this.ganIndex = yearGanIndex;
        this.zhiIndex = yearZhiIndex;
        compute();
    }


    public static LunarYear fromYear(int lunarYear)
    {
        LunarYear y;
        if (LunarYear.CACHE_YEAR == null || LunarYear.CACHE_YEAR.getYear() != lunarYear) {
            y = new LunarYear(lunarYear);
            LunarYear.CACHE_YEAR = y;
        } else {
            y = LunarYear.CACHE_YEAR;
        }
        return y;
    }

    @Override
    public String toString() {
        return year+"";
    }


    public String toFullString()
    {
        return year + "年";
    }

    public int getYear()
    {
        return year;
    }

    public int getGanIndex()
    {
        return ganIndex;
    }

    public int getZhiIndex()
    {
        return zhiIndex;
    }

    public String getGan()
    {
        return LunarUtil.GAN.get(ganIndex + 1);
    }

    public String getZhi()
    {
        return LunarUtil.ZHI.get(zhiIndex + 1);
    }

    public String getGanZhi()
    {
        return getGan() + getZhi();
    }

    public List<Double> getJieQiJulianDays()
    {
        return jieQiJulianDays;
    }

    public List<LunarMonth> getMonths()
    {
        return months;
    }

    public int getDayCount()
    {
        int n = 0;
        for (LunarMonth month : months) {
            if(month.getYear() == year){
                n += month.getDayCount();
            }
        }
        return n;
    }

    public List<LunarMonth> getMonthsInYear()
    {
        List<LunarMonth> l = new ArrayList<>();
        for (LunarMonth month : months) {
            if(month.getYear() == year){
                l.add(month);
            }
        }
        return l;
    }

    public LunarMonth getMonth(int lunarMonth)
    {
        for (int i = 0; i < months.size(); i++) {
            LunarMonth m = months.get(i);
            if(m.getYear() == year && m.getMonth() == lunarMonth){
                return m;
            }
        }
        return null;
    }

    public int getLeapMonth()
    {
        for (LunarMonth month : months) {
            if(month.getYear() == year && month.isLeap()){
                return Math.abs(month.getMonth());
            }
        }
        return 0;
    }

    private void compute()
    {
        List<Double> jq = new ArrayList<>();
        List<Double> hs = new ArrayList<>();
        List<Integer> dayCounts = new ArrayList<>();
        List<Integer> months = new ArrayList<>();
        int currentYear = year;
        double jd = Math.floor((currentYear - 2000) * 365.2422 + 180);
        double w = Math.floor((jd - 355 + 183) / 365.2422) * 365.2422 + 355;
        if (ShouXingUtil.calcQi(w) > jd) {
            w -= 365.2422;
        }
        for (int i = 0; i < 26; i++) {
            jq.add(ShouXingUtil.calcQi(w + 15.2184 * i));
        }
        for (int i = 0, j = Lunar.JIE_QI_IN_USE.size(); i < j; i++) {
            if (i == 0) {
                jd = ShouXingUtil.qiAccurate2(jq.get(0) - 15.2184);
            } else if (i <= 26) {
                jd = ShouXingUtil.qiAccurate2(jq.get(i - 1));
            } else {
                jd = ShouXingUtil.qiAccurate2(jq.get(25) + 15.2184 * (i - 26));
            }
            jieQiJulianDays.add( (jd + Solar.J2000));
        }
        w = ShouXingUtil.calcShuo(jq.get(0));
        if (w > jq.get(0)) {
            w -= 29.53;
        }
        for (int i = 0; i < 16; i++) {
            hs.add(ShouXingUtil.calcShuo(w + 29.5306 * i));
        }
        int i;
        for (i = 0; i < 15; i++) {
            dayCounts.add((int) (hs.get(i + 1) - hs.get(i)));
            months.add(i);
        }
        int prevYear = currentYear - 1;
        int leapIndex = 16;
        if(LEAP_11.contains(currentYear)){
            leapIndex = 13;
        }else if(LEAP_12.contains(currentYear)){
            leapIndex = 14;
        }else if(hs.get(13) <= jq.get(24)){
            i = 1;
            while (hs.get(i + 1) > jq.get(2 * i) && i < 13) {
                i++;
            }
            leapIndex = i;
        }
        for (i = leapIndex; i < 15; i++) {
            int i1 = months.get(i) - 1;
            months.set(i, i1);
        }
        int fm = -1;
        int index = -1;
        int y = prevYear;
        for (i = 0; i < 15; i++) {
            double dm = hs.get(i) + Solar.J2000;
            int v2 = months.get(i);
            int mc = YMC.get(v2 % 12);
            if (1724360 <= dm && dm < 1729794) {
                mc = YMC.get((v2 + 1) % 12);
            } else if (1807724 <= dm && dm < 1808699) {
                mc = YMC.get((v2 + 1) % 12);
            } else if (dm == 1729794 || dm == 1808699) {
                mc = 12;
            }
            if (fm == -1) {
                fm = mc;
                index = mc;
            }
            if (mc < fm) {
                y += 1;
                index = 1;
            }
            fm = mc;
            if (i == leapIndex) {
                mc = -mc;
            } else if (dm == 1729794 || dm == 1808699) {
                mc = -11;
            }
            this.months.add(new LunarMonth(y, mc, dayCounts.get(i), (int) (hs.get(i) + Solar.J2000), index));
            index++;
        }
    }

    protected String getZaoByGan(int index, String name)
    {
        LunarMonth month = getMonth(1);
        if (null == month) {
            return "";
        }
        int offset = index - Solar.fromJulianDay(month.getFirstJulianDay()).getLunar().getDayGanIndex();
        if (offset < 0) {
            offset += 10;
        }
        return name.replace("几", LunarUtil.NUMBER.get(offset + 1));
    }

    protected String getZaoByZhi(int index,String name)
    {
        LunarMonth month = getMonth(1);
        if (null == month) {
            return "";
        }
        int offset = index - Solar.fromJulianDay(month.getFirstJulianDay()).getLunar().getDayZhiIndex();
        if (offset < 0) {
            offset += 12;
        }
        return name.replace("几", LunarUtil.NUMBER.get(offset + 1));
    }

    public String getTouLiang()
    {
        return getZaoByZhi(0, "几鼠偷粮");
    }

    public String getCaoZi()
    {
        return getZaoByZhi(0, "草子几分");
    }

    public String getGengTian()
    {
        return getZaoByZhi(1, "几牛耕田");
    }

    public String getHuaShou()
    {
        return getZaoByZhi(3, "花收几分");
    }

    public String getZhiShui()
    {
        return getZaoByZhi(4, "几龙治水");
    }

    public String getTuoGu()
    {
        return getZaoByZhi(6, "几马驮谷");
    }

    public String getQiangMi()
    {
        return getZaoByZhi(9, "几鸡抢米");
    }

    public String getKanCan()
    {
        return getZaoByZhi(9, "几姑看蚕");
    }

    public String getGongZhu()
    {
        return getZaoByZhi(11, "几屠共猪");
    }

    public String getJiaTian()
    {
        return getZaoByGan(0, "甲田几分");
    }

    public String getFenBing()
    {
        return getZaoByGan(2, "几人分饼");
    }

    public String getDeJin()
    {
        return getZaoByGan(7, "几日得金");
    }

    public String getRenBing()
    {
        return getZaoByGan(2, getZaoByZhi(2, "几人几丙"));
    }

    public String getRenChu()
    {
        return getZaoByGan(3, getZaoByZhi(2, "几人几锄"));
    }

    public String getYuan()
    {
        return LunarYear.YUAN.get(((year + 2696) / 60) % 3) + "元";
    }

    public String getYun()
    {
        return LunarYear.YUN.get(((year + 2696) / 20) % 9) + "运";
    }

    public NineStar getNineStar()
    {
        int index = LunarUtil.getJiaZiIndex(getGanZhi()) + 1;
        int yuan = ((year + 2696) / 60) % 3;
        int offset = (62 + yuan * 3 - index) % 9;
        if (0 == offset) {
            offset = 9;
        }
        return NineStar.fromIndex(offset - 1);
    }

    public String getPositionXi()
    {
        return LunarUtil.POSITION_XI.get(ganIndex + 1);
    }

    public String getPositionXiDesc()
    {
        return LunarUtil.POSITION_DESC.get(getPositionXi());
    }

    public String getPositionYangGui()
    {
        return LunarUtil.POSITION_YANG_GUI.get(ganIndex + 1);
    }

    public String getPositionYangGuiDesc()
    {
        return LunarUtil.POSITION_DESC.get(getPositionYangGui());
    }

    public String getPositionYinGui()
    {
        return LunarUtil.POSITION_YIN_GUI.get(ganIndex + 1);
    }

    public String getPositionYinGuiDesc()
    {
        return LunarUtil.POSITION_DESC.get(getPositionYinGui());
    }

    public String getPositionFu()
    {
        return getPositionFuBySect(2);
    }

    public String getPositionFuBySect(int sect)
    {
        List<String> fu = 1 == sect ? LunarUtil.POSITION_FU : LunarUtil.POSITION_FU_2;
        return fu.get(ganIndex + 1);
    }

    public String getPositionFuDesc()
    {
        return getPositionFuDescBySect(2);
    }

    public String getPositionFuDescBySect(int sect)
    {
        return LunarUtil.POSITION_DESC.get(getPositionFuBySect(sect));
    }

    public String getPositionCai()
    {
        return LunarUtil.POSITION_CAI.get(ganIndex + 1);
    }

    public String getPositionCaiDesc()
    {
        return LunarUtil.POSITION_DESC.get(getPositionCai());
    }

    public String getPositionTaiSui()
    {
        return LunarUtil.POSITION_TAI_SUI_YEAR.get(zhiIndex);
    }

    public String getPositionTaiSuiDesc()
    {
        return LunarUtil.POSITION_DESC.get(getPositionTaiSui());
    }

    public LunarYear next(int n)
    {
        return LunarYear.fromYear(year + n);
    }

    public static void main(String[] args) {
        System.out.println(LunarYear.fromYear(2021));
    }
}
